home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / urllib2.pyc (.txt) < prev    next >
Python Compiled Bytecode  |  2005-10-18  |  41KB  |  1,309 lines

  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. '''An extensible library for opening URLs using a variety of protocols
  5.  
  6. The simplest way to use this module is to call the urlopen function,
  7. which accepts a string containing a URL or a Request object (described
  8. below).  It opens the URL and returns the results as file-like
  9. object; the returned object has some extra methods described below.
  10.  
  11. The OpenerDirector manages a collection of Handler objects that do
  12. all the actual work.  Each Handler implements a particular protocol or
  13. option.  The OpenerDirector is a composite object that invokes the
  14. Handlers needed to open the requested URL.  For example, the
  15. HTTPHandler performs HTTP GET and POST requests and deals with
  16. non-error returns.  The HTTPRedirectHandler automatically deals with
  17. HTTP 301, 302, 303 and 307 redirect errors, and the HTTPDigestAuthHandler
  18. deals with digest authentication.
  19.  
  20. urlopen(url, data=None) -- basic usage is that same as original
  21. urllib.  pass the url and optionally data to post to an HTTP URL, and
  22. get a file-like object back.  One difference is that you can also pass
  23. a Request instance instead of URL.  Raises a URLError (subclass of
  24. IOError); for HTTP errors, raises an HTTPError, which can also be
  25. treated as a valid response.
  26.  
  27. build_opener -- function that creates a new OpenerDirector instance.
  28. will install the default handlers.  accepts one or more Handlers as
  29. arguments, either instances or Handler classes that it will
  30. instantiate.  if one of the argument is a subclass of the default
  31. handler, the argument will be installed instead of the default.
  32.  
  33. install_opener -- installs a new opener as the default opener.
  34.  
  35. objects of interest:
  36. OpenerDirector --
  37.  
  38. Request -- an object that encapsulates the state of a request.  the
  39. state can be a simple as the URL.  it can also include extra HTTP
  40. headers, e.g. a User-Agent.
  41.  
  42. BaseHandler --
  43.  
  44. exceptions:
  45. URLError-- a subclass of IOError, individual protocols have their own
  46. specific subclass
  47.  
  48. HTTPError-- also a valid HTTP response, so you can treat an HTTP error
  49. as an exceptional event or valid response
  50.  
  51. internals:
  52. BaseHandler and parent
  53. _call_chain conventions
  54.  
  55. Example usage:
  56.  
  57. import urllib2
  58.  
  59. # set up authentication info
  60. authinfo = urllib2.HTTPBasicAuthHandler()
  61. authinfo.add_password(\'realm\', \'host\', \'username\', \'password\')
  62.  
  63. proxy_support = urllib2.ProxyHandler({"http" : "http://ahad-haam:3128"})
  64.  
  65. # build a new opener that adds authentication and caching FTP handlers
  66. opener = urllib2.build_opener(proxy_support, authinfo, urllib2.CacheFTPHandler)
  67.  
  68. # install it
  69. urllib2.install_opener(opener)
  70.  
  71. f = urllib2.urlopen(\'http://www.python.org/\')
  72.  
  73.  
  74. '''
  75. import base64
  76. import ftplib
  77. import gopherlib
  78. import httplib
  79. import inspect
  80. import md5
  81. import mimetypes
  82. import mimetools
  83. import os
  84. import posixpath
  85. import random
  86. import re
  87. import sha
  88. import socket
  89. import sys
  90. import time
  91. import urlparse
  92. import bisect
  93. import cookielib
  94.  
  95. try:
  96.     from cStringIO import StringIO
  97. except ImportError:
  98.     from StringIO import StringIO
  99.  
  100. from urllib import unwrap, unquote, splittype, splithost, addinfourl, splitport, splitgophertype, splitquery, splitattr, ftpwrapper, noheaders, splituser, splitpasswd, splitvalue
  101. from urllib import localhost, url2pathname, getproxies
  102. __version__ = '2.4'
  103. _opener = None
  104.  
  105. def urlopen(url, data = None):
  106.     global _opener
  107.     if _opener is None:
  108.         _opener = build_opener()
  109.     
  110.     return _opener.open(url, data)
  111.  
  112.  
  113. def install_opener(opener):
  114.     global _opener
  115.     _opener = opener
  116.  
  117.  
  118. class URLError(IOError):
  119.     
  120.     def __init__(self, reason):
  121.         self.args = (reason,)
  122.         self.reason = reason
  123.  
  124.     
  125.     def __str__(self):
  126.         return '<urlopen error %s>' % self.reason
  127.  
  128.  
  129.  
  130. class HTTPError(URLError, addinfourl):
  131.     '''Raised when HTTP error occurs, but also acts like non-error return'''
  132.     __super_init = addinfourl.__init__
  133.     
  134.     def __init__(self, url, code, msg, hdrs, fp):
  135.         self.code = code
  136.         self.msg = msg
  137.         self.hdrs = hdrs
  138.         self.fp = fp
  139.         self.filename = url
  140.         if fp is not None:
  141.             self._HTTPError__super_init(fp, hdrs, url)
  142.         
  143.  
  144.     
  145.     def __str__(self):
  146.         return 'HTTP Error %s: %s' % (self.code, self.msg)
  147.  
  148.  
  149.  
  150. class GopherError(URLError):
  151.     pass
  152.  
  153.  
  154. class Request:
  155.     
  156.     def __init__(self, url, data = None, headers = { }, origin_req_host = None, unverifiable = False):
  157.         self._Request__original = unwrap(url)
  158.         self.type = None
  159.         self.host = None
  160.         self.port = None
  161.         self.data = data
  162.         self.headers = { }
  163.         for key, value in headers.items():
  164.             self.add_header(key, value)
  165.         
  166.         self.unredirected_hdrs = { }
  167.         if origin_req_host is None:
  168.             origin_req_host = cookielib.request_host(self)
  169.         
  170.         self.origin_req_host = origin_req_host
  171.         self.unverifiable = unverifiable
  172.  
  173.     
  174.     def __getattr__(self, attr):
  175.         if attr[:12] == '_Request__r_':
  176.             name = attr[12:]
  177.             if hasattr(Request, 'get_' + name):
  178.                 getattr(self, 'get_' + name)()
  179.                 return getattr(self, attr)
  180.             
  181.         
  182.         raise AttributeError, attr
  183.  
  184.     
  185.     def get_method(self):
  186.         if self.has_data():
  187.             return 'POST'
  188.         else:
  189.             return 'GET'
  190.  
  191.     
  192.     def add_data(self, data):
  193.         self.data = data
  194.  
  195.     
  196.     def has_data(self):
  197.         return self.data is not None
  198.  
  199.     
  200.     def get_data(self):
  201.         return self.data
  202.  
  203.     
  204.     def get_full_url(self):
  205.         return self._Request__original
  206.  
  207.     
  208.     def get_type(self):
  209.         if self.type is None:
  210.             (self.type, self._Request__r_type) = splittype(self._Request__original)
  211.             if self.type is None:
  212.                 raise ValueError, 'unknown url type: %s' % self._Request__original
  213.             
  214.         
  215.         return self.type
  216.  
  217.     
  218.     def get_host(self):
  219.         if self.host is None:
  220.             (self.host, self._Request__r_host) = splithost(self._Request__r_type)
  221.             if self.host:
  222.                 self.host = unquote(self.host)
  223.             
  224.         
  225.         return self.host
  226.  
  227.     
  228.     def get_selector(self):
  229.         return self._Request__r_host
  230.  
  231.     
  232.     def set_proxy(self, host, type):
  233.         self.host = host
  234.         self.type = type
  235.         self._Request__r_host = self._Request__original
  236.  
  237.     
  238.     def get_origin_req_host(self):
  239.         return self.origin_req_host
  240.  
  241.     
  242.     def is_unverifiable(self):
  243.         return self.unverifiable
  244.  
  245.     
  246.     def add_header(self, key, val):
  247.         self.headers[key.capitalize()] = val
  248.  
  249.     
  250.     def add_unredirected_header(self, key, val):
  251.         self.unredirected_hdrs[key.capitalize()] = val
  252.  
  253.     
  254.     def has_header(self, header_name):
  255.         if not header_name in self.headers:
  256.             pass
  257.         return header_name in self.unredirected_hdrs
  258.  
  259.     
  260.     def get_header(self, header_name, default = None):
  261.         return self.headers.get(header_name, self.unredirected_hdrs.get(header_name, default))
  262.  
  263.     
  264.     def header_items(self):
  265.         hdrs = self.unredirected_hdrs.copy()
  266.         hdrs.update(self.headers)
  267.         return hdrs.items()
  268.  
  269.  
  270.  
  271. class OpenerDirector:
  272.     
  273.     def __init__(self):
  274.         server_version = 'Python-urllib/%s' % __version__
  275.         self.addheaders = [
  276.             ('User-agent', server_version)]
  277.         self.handlers = []
  278.         self.handle_open = { }
  279.         self.handle_error = { }
  280.         self.process_response = { }
  281.         self.process_request = { }
  282.  
  283.     
  284.     def add_handler(self, handler):
  285.         added = False
  286.         for meth in dir(handler):
  287.             i = meth.find('_')
  288.             protocol = meth[:i]
  289.             condition = meth[i + 1:]
  290.             if condition.startswith('error'):
  291.                 j = condition.find('_') + i + 1
  292.                 kind = meth[j + 1:]
  293.                 
  294.                 try:
  295.                     kind = int(kind)
  296.                 except ValueError:
  297.                     pass
  298.  
  299.                 lookup = self.handle_error.get(protocol, { })
  300.                 self.handle_error[protocol] = lookup
  301.             elif condition == 'open':
  302.                 kind = protocol
  303.                 lookup = getattr(self, 'handle_' + condition)
  304.             elif condition in [
  305.                 'response',
  306.                 'request']:
  307.                 kind = protocol
  308.                 lookup = getattr(self, 'process_' + condition)
  309.             
  310.             handlers = lookup.setdefault(kind, [])
  311.             if handlers:
  312.                 bisect.insort(handlers, handler)
  313.             else:
  314.                 handlers.append(handler)
  315.             added = True
  316.         
  317.         if added:
  318.             bisect.insort(self.handlers, handler)
  319.             handler.add_parent(self)
  320.         
  321.  
  322.     
  323.     def close(self):
  324.         pass
  325.  
  326.     
  327.     def _call_chain(self, chain, kind, meth_name, *args):
  328.         handlers = chain.get(kind, ())
  329.         for handler in handlers:
  330.             func = getattr(handler, meth_name)
  331.             result = func(*args)
  332.             if result is not None:
  333.                 return result
  334.                 continue
  335.         
  336.  
  337.     
  338.     def open(self, fullurl, data = None):
  339.         if isinstance(fullurl, basestring):
  340.             req = Request(fullurl, data)
  341.         else:
  342.             req = fullurl
  343.             if data is not None:
  344.                 req.add_data(data)
  345.             
  346.         protocol = req.get_type()
  347.         meth_name = protocol + '_request'
  348.         for processor in self.process_request.get(protocol, []):
  349.             meth = getattr(processor, meth_name)
  350.             req = meth(req)
  351.         
  352.         response = self._open(req, data)
  353.         meth_name = protocol + '_response'
  354.         for processor in self.process_response.get(protocol, []):
  355.             meth = getattr(processor, meth_name)
  356.             response = meth(req, response)
  357.         
  358.         return response
  359.  
  360.     
  361.     def _open(self, req, data = None):
  362.         result = self._call_chain(self.handle_open, 'default', 'default_open', req)
  363.         if result:
  364.             return result
  365.         
  366.         protocol = req.get_type()
  367.         result = self._call_chain(self.handle_open, protocol, protocol + '_open', req)
  368.         if result:
  369.             return result
  370.         
  371.         return self._call_chain(self.handle_open, 'unknown', 'unknown_open', req)
  372.  
  373.     
  374.     def error(self, proto, *args):
  375.         if proto in [
  376.             'http',
  377.             'https']:
  378.             dict = self.handle_error['http']
  379.             proto = args[2]
  380.             meth_name = 'http_error_%s' % proto
  381.             http_err = 1
  382.             orig_args = args
  383.         else:
  384.             dict = self.handle_error
  385.             meth_name = proto + '_error'
  386.             http_err = 0
  387.         args = (dict, proto, meth_name) + args
  388.         result = self._call_chain(*args)
  389.         if result:
  390.             return result
  391.         
  392.         if http_err:
  393.             args = (dict, 'default', 'http_error_default') + orig_args
  394.             return self._call_chain(*args)
  395.         
  396.  
  397.  
  398.  
  399. def build_opener(*handlers):
  400.     '''Create an opener object from a list of handlers.
  401.  
  402.     The opener will use several default handlers, including support
  403.     for HTTP and FTP.
  404.  
  405.     If any of the handlers passed as arguments are subclasses of the
  406.     default handlers, the default handlers will not be used.
  407.     '''
  408.     opener = OpenerDirector()
  409.     default_classes = [
  410.         ProxyHandler,
  411.         UnknownHandler,
  412.         HTTPHandler,
  413.         HTTPDefaultErrorHandler,
  414.         HTTPRedirectHandler,
  415.         FTPHandler,
  416.         FileHandler,
  417.         HTTPErrorProcessor]
  418.     if hasattr(httplib, 'HTTPS'):
  419.         default_classes.append(HTTPSHandler)
  420.     
  421.     skip = []
  422.     for klass in default_classes:
  423.         for check in handlers:
  424.             if inspect.isclass(check):
  425.                 if issubclass(check, klass):
  426.                     skip.append(klass)
  427.                 
  428.             issubclass(check, klass)
  429.             if isinstance(check, klass):
  430.                 skip.append(klass)
  431.                 continue
  432.         
  433.     
  434.     for klass in skip:
  435.         default_classes.remove(klass)
  436.     
  437.     for klass in default_classes:
  438.         opener.add_handler(klass())
  439.     
  440.     for h in handlers:
  441.         if inspect.isclass(h):
  442.             h = h()
  443.         
  444.         opener.add_handler(h)
  445.     
  446.     return opener
  447.  
  448.  
  449. class BaseHandler:
  450.     handler_order = 500
  451.     
  452.     def add_parent(self, parent):
  453.         self.parent = parent
  454.  
  455.     
  456.     def close(self):
  457.         pass
  458.  
  459.     
  460.     def __lt__(self, other):
  461.         if not hasattr(other, 'handler_order'):
  462.             return True
  463.         
  464.         return self.handler_order < other.handler_order
  465.  
  466.  
  467.  
  468. class HTTPErrorProcessor(BaseHandler):
  469.     '''Process HTTP error responses.'''
  470.     handler_order = 1000
  471.     
  472.     def http_response(self, request, response):
  473.         code = response.code
  474.         msg = response.msg
  475.         hdrs = response.info()
  476.         if code not in (200, 206):
  477.             response = self.parent.error('http', request, response, code, msg, hdrs)
  478.         
  479.         return response
  480.  
  481.     https_response = http_response
  482.  
  483.  
  484. class HTTPDefaultErrorHandler(BaseHandler):
  485.     
  486.     def http_error_default(self, req, fp, code, msg, hdrs):
  487.         raise HTTPError(req.get_full_url(), code, msg, hdrs, fp)
  488.  
  489.  
  490.  
  491. class HTTPRedirectHandler(BaseHandler):
  492.     max_repeats = 4
  493.     max_redirections = 10
  494.     
  495.     def redirect_request(self, req, fp, code, msg, headers, newurl):
  496.         """Return a Request or None in response to a redirect.
  497.  
  498.         This is called by the http_error_30x methods when a
  499.         redirection response is received.  If a redirection should
  500.         take place, return a new Request to allow http_error_30x to
  501.         perform the redirect.  Otherwise, raise HTTPError if no-one
  502.         else should try to handle this url.  Return None if you can't
  503.         but another Handler might.
  504.         """
  505.         m = req.get_method()
  506.         if (code in (301, 302, 303, 307) or m in ('GET', 'HEAD') or code in (301, 302, 303)) and m == 'POST':
  507.             return Request(newurl, headers = req.headers, origin_req_host = req.get_origin_req_host(), unverifiable = True)
  508.         else:
  509.             raise HTTPError(req.get_full_url(), code, msg, headers, fp)
  510.  
  511.     
  512.     def http_error_302(self, req, fp, code, msg, headers):
  513.         if 'location' in headers:
  514.             newurl = headers.getheaders('location')[0]
  515.         elif 'uri' in headers:
  516.             newurl = headers.getheaders('uri')[0]
  517.         else:
  518.             return None
  519.         newurl = urlparse.urljoin(req.get_full_url(), newurl)
  520.         new = self.redirect_request(req, fp, code, msg, headers, newurl)
  521.         if new is None:
  522.             return None
  523.         
  524.         if hasattr(req, 'redirect_dict'):
  525.             visited = new.redirect_dict = req.redirect_dict
  526.             if visited.get(newurl, 0) >= self.max_repeats or len(visited) >= self.max_redirections:
  527.                 raise HTTPError(req.get_full_url(), code, self.inf_msg + msg, headers, fp)
  528.             
  529.         else:
  530.             visited = new.redirect_dict = req.redirect_dict = { }
  531.         visited[newurl] = visited.get(newurl, 0) + 1
  532.         fp.read()
  533.         fp.close()
  534.         return self.parent.open(new)
  535.  
  536.     http_error_301 = http_error_303 = http_error_307 = http_error_302
  537.     inf_msg = 'The HTTP server returned a redirect error that would lead to an infinite loop.\nThe last 30x error message was:\n'
  538.  
  539.  
  540. class ProxyHandler(BaseHandler):
  541.     handler_order = 100
  542.     
  543.     def __init__(self, proxies = None):
  544.         if proxies is None:
  545.             proxies = getproxies()
  546.         
  547.         if not hasattr(proxies, 'has_key'):
  548.             raise AssertionError, 'proxies must be a mapping'
  549.         self.proxies = proxies
  550.         for type, url in proxies.items():
  551.             setattr(self, '%s_open' % type, (lambda r, proxy = url, type = type, meth = self.proxy_open: meth(r, proxy, type)))
  552.         
  553.  
  554.     
  555.     def proxy_open(self, req, proxy, type):
  556.         orig_type = req.get_type()
  557.         (type, r_type) = splittype(proxy)
  558.         (host, XXX) = splithost(r_type)
  559.         if '@' in host:
  560.             (user_pass, host) = host.split('@', 1)
  561.             if ':' in user_pass:
  562.                 (user, password) = user_pass.split(':', 1)
  563.                 user_pass = base64.encodestring('%s:%s' % (unquote(user), unquote(password))).strip()
  564.                 req.add_header('Proxy-authorization', 'Basic ' + user_pass)
  565.             
  566.         
  567.         host = unquote(host)
  568.         req.set_proxy(host, type)
  569.         if orig_type == type:
  570.             return None
  571.         else:
  572.             return self.parent.open(req)
  573.  
  574.  
  575.  
  576. class CustomProxy:
  577.     
  578.     def __init__(self, proto, func = None, proxy_addr = None):
  579.         self.proto = proto
  580.         self.func = func
  581.         self.addr = proxy_addr
  582.  
  583.     
  584.     def handle(self, req):
  585.         if self.func and self.func(req):
  586.             return 1
  587.         
  588.  
  589.     
  590.     def get_proxy(self):
  591.         return self.addr
  592.  
  593.  
  594.  
  595. class CustomProxyHandler(BaseHandler):
  596.     handler_order = 100
  597.     
  598.     def __init__(self, *proxies):
  599.         self.proxies = { }
  600.  
  601.     
  602.     def proxy_open(self, req):
  603.         proto = req.get_type()
  604.         
  605.         try:
  606.             proxies = self.proxies[proto]
  607.         except KeyError:
  608.             return None
  609.  
  610.         for p in proxies:
  611.             if p.handle(req):
  612.                 req.set_proxy(p.get_proxy())
  613.                 return self.parent.open(req)
  614.                 continue
  615.         
  616.  
  617.     
  618.     def do_proxy(self, p, req):
  619.         return self.parent.open(req)
  620.  
  621.     
  622.     def add_proxy(self, cpo):
  623.         if cpo.proto in self.proxies:
  624.             self.proxies[cpo.proto].append(cpo)
  625.         else:
  626.             self.proxies[cpo.proto] = [
  627.                 cpo]
  628.  
  629.  
  630.  
  631. class HTTPPasswordMgr:
  632.     
  633.     def __init__(self):
  634.         self.passwd = { }
  635.  
  636.     
  637.     def add_password(self, realm, uri, user, passwd):
  638.         if isinstance(uri, basestring):
  639.             uri = [
  640.                 uri]
  641.         
  642.         uri = tuple(map(self.reduce_uri, uri))
  643.         if realm not in self.passwd:
  644.             self.passwd[realm] = { }
  645.         
  646.         self.passwd[realm][uri] = (user, passwd)
  647.  
  648.     
  649.     def find_user_password(self, realm, authuri):
  650.         domains = self.passwd.get(realm, { })
  651.         authuri = self.reduce_uri(authuri)
  652.         for uris, authinfo in domains.iteritems():
  653.             for uri in uris:
  654.                 if self.is_suburi(uri, authuri):
  655.                     return authinfo
  656.                     continue
  657.             
  658.         
  659.         return (None, None)
  660.  
  661.     
  662.     def reduce_uri(self, uri):
  663.         '''Accept netloc or URI and extract only the netloc and path'''
  664.         parts = urlparse.urlparse(uri)
  665.         if parts[1]:
  666.             if not parts[2]:
  667.                 pass
  668.             return (parts[1], '/')
  669.         else:
  670.             return (parts[2], '/')
  671.  
  672.     
  673.     def is_suburi(self, base, test):
  674.         '''Check if test is below base in a URI tree
  675.  
  676.         Both args must be URIs in reduced form.
  677.         '''
  678.         if base == test:
  679.             return True
  680.         
  681.         if base[0] != test[0]:
  682.             return False
  683.         
  684.         common = posixpath.commonprefix((base[1], test[1]))
  685.         if len(common) == len(base[1]):
  686.             return True
  687.         
  688.         return False
  689.  
  690.  
  691.  
  692. class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
  693.     
  694.     def find_user_password(self, realm, authuri):
  695.         (user, password) = HTTPPasswordMgr.find_user_password(self, realm, authuri)
  696.         if user is not None:
  697.             return (user, password)
  698.         
  699.         return HTTPPasswordMgr.find_user_password(self, None, authuri)
  700.  
  701.  
  702.  
  703. class AbstractBasicAuthHandler:
  704.     rx = re.compile('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', re.I)
  705.     
  706.     def __init__(self, password_mgr = None):
  707.         if password_mgr is None:
  708.             password_mgr = HTTPPasswordMgr()
  709.         
  710.         self.passwd = password_mgr
  711.         self.add_password = self.passwd.add_password
  712.  
  713.     
  714.     def http_error_auth_reqed(self, authreq, host, req, headers):
  715.         authreq = headers.get(authreq, None)
  716.         if authreq:
  717.             mo = AbstractBasicAuthHandler.rx.search(authreq)
  718.             if mo:
  719.                 (scheme, realm) = mo.groups()
  720.                 if scheme.lower() == 'basic':
  721.                     return self.retry_http_basic_auth(host, req, realm)
  722.                 
  723.             
  724.         
  725.  
  726.     
  727.     def retry_http_basic_auth(self, host, req, realm):
  728.         (user, pw) = self.passwd.find_user_password(realm, host)
  729.         if pw is not None:
  730.             raw = '%s:%s' % (user, pw)
  731.             auth = 'Basic %s' % base64.encodestring(raw).strip()
  732.             if req.headers.get(self.auth_header, None) == auth:
  733.                 return None
  734.             
  735.             req.add_header(self.auth_header, auth)
  736.             return self.parent.open(req)
  737.         else:
  738.             return None
  739.  
  740.  
  741.  
  742. class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
  743.     auth_header = 'Authorization'
  744.     
  745.     def http_error_401(self, req, fp, code, msg, headers):
  746.         host = urlparse.urlparse(req.get_full_url())[1]
  747.         return self.http_error_auth_reqed('www-authenticate', host, req, headers)
  748.  
  749.  
  750.  
  751. class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
  752.     auth_header = 'Proxy-authorization'
  753.     
  754.     def http_error_407(self, req, fp, code, msg, headers):
  755.         host = req.get_host()
  756.         return self.http_error_auth_reqed('proxy-authenticate', host, req, headers)
  757.  
  758.  
  759.  
  760. def randombytes(n):
  761.     '''Return n random bytes.'''
  762.     pass
  763.  
  764.  
  765. class AbstractDigestAuthHandler:
  766.     
  767.     def __init__(self, passwd = None):
  768.         if passwd is None:
  769.             passwd = HTTPPasswordMgr()
  770.         
  771.         self.passwd = passwd
  772.         self.add_password = self.passwd.add_password
  773.         self.retried = 0
  774.         self.nonce_count = 0
  775.  
  776.     
  777.     def reset_retry_count(self):
  778.         self.retried = 0
  779.  
  780.     
  781.     def http_error_auth_reqed(self, auth_header, host, req, headers):
  782.         authreq = headers.get(auth_header, None)
  783.         if authreq:
  784.             scheme = authreq.split()[0]
  785.             if scheme.lower() == 'digest':
  786.                 return self.retry_http_digest_auth(req, authreq)
  787.             else:
  788.                 raise ValueError("AbstractDigestAuthHandler doesn't know about %s" % scheme)
  789.         
  790.  
  791.     
  792.     def retry_http_digest_auth(self, req, auth):
  793.         (token, challenge) = auth.split(' ', 1)
  794.         chal = parse_keqv_list(parse_http_list(challenge))
  795.         auth = self.get_authorization(req, chal)
  796.         if auth:
  797.             auth_val = 'Digest %s' % auth
  798.             if req.headers.get(self.auth_header, None) == auth_val:
  799.                 return None
  800.             
  801.             req.add_header(self.auth_header, auth_val)
  802.             resp = self.parent.open(req)
  803.             return resp
  804.         
  805.  
  806.     
  807.     def get_cnonce(self, nonce):
  808.         dig = sha.new('%s:%s:%s:%s' % (self.nonce_count, nonce, time.ctime(), randombytes(8))).hexdigest()
  809.         return dig[:16]
  810.  
  811.     
  812.     def get_authorization(self, req, chal):
  813.         
  814.         try:
  815.             realm = chal['realm']
  816.             nonce = chal['nonce']
  817.             qop = chal.get('qop')
  818.             algorithm = chal.get('algorithm', 'MD5')
  819.             opaque = chal.get('opaque', None)
  820.         except KeyError:
  821.             return None
  822.  
  823.         (H, KD) = self.get_algorithm_impls(algorithm)
  824.         if H is None:
  825.             return None
  826.         
  827.         (user, pw) = self.passwd.find_user_password(realm, req.get_full_url())
  828.         if user is None:
  829.             return None
  830.         
  831.         if req.has_data():
  832.             entdig = self.get_entity_digest(req.get_data(), chal)
  833.         else:
  834.             entdig = None
  835.         A1 = '%s:%s:%s' % (user, realm, pw)
  836.         A2 = '%s:%s' % (req.get_method(), req.get_selector())
  837.         if qop == 'auth':
  838.             self.nonce_count += 1
  839.             ncvalue = '%08x' % self.nonce_count
  840.             cnonce = self.get_cnonce(nonce)
  841.             noncebit = '%s:%s:%s:%s:%s' % (nonce, ncvalue, cnonce, qop, H(A2))
  842.             respdig = KD(H(A1), noncebit)
  843.         elif qop is None:
  844.             respdig = KD(H(A1), '%s:%s' % (nonce, H(A2)))
  845.         
  846.         base = 'username="%s", realm="%s", nonce="%s", uri="%s", response="%s"' % (user, realm, nonce, req.get_selector(), respdig)
  847.         if opaque:
  848.             base = base + ', opaque="%s"' % opaque
  849.         
  850.         if entdig:
  851.             base = base + ', digest="%s"' % entdig
  852.         
  853.         if algorithm != 'MD5':
  854.             base = base + ', algorithm="%s"' % algorithm
  855.         
  856.         if qop:
  857.             base = base + ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
  858.         
  859.         return base
  860.  
  861.     
  862.     def get_algorithm_impls(self, algorithm):
  863.         if algorithm == 'MD5':
  864.             
  865.             H = lambda x: md5.new(x).hexdigest()
  866.         elif algorithm == 'SHA':
  867.             
  868.             H = lambda x: sha.new(x).hexdigest()
  869.         
  870.         
  871.         KD = lambda s, d: H('%s:%s' % (s, d))
  872.         return (H, KD)
  873.  
  874.     
  875.     def get_entity_digest(self, data, chal):
  876.         pass
  877.  
  878.  
  879.  
  880. class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
  881.     '''An authentication protocol defined by RFC 2069
  882.  
  883.     Digest authentication improves on basic authentication because it
  884.     does not transmit passwords in the clear.
  885.     '''
  886.     auth_header = 'Authorization'
  887.     
  888.     def http_error_401(self, req, fp, code, msg, headers):
  889.         host = urlparse.urlparse(req.get_full_url())[1]
  890.         retry = self.http_error_auth_reqed('www-authenticate', host, req, headers)
  891.         self.reset_retry_count()
  892.         return retry
  893.  
  894.  
  895.  
  896. class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
  897.     auth_header = 'Proxy-Authorization'
  898.     
  899.     def http_error_407(self, req, fp, code, msg, headers):
  900.         host = req.get_host()
  901.         retry = self.http_error_auth_reqed('proxy-authenticate', host, req, headers)
  902.         self.reset_retry_count()
  903.         return retry
  904.  
  905.  
  906.  
  907. class AbstractHTTPHandler(BaseHandler):
  908.     
  909.     def __init__(self, debuglevel = 0):
  910.         self._debuglevel = debuglevel
  911.  
  912.     
  913.     def set_http_debuglevel(self, level):
  914.         self._debuglevel = level
  915.  
  916.     
  917.     def do_request_(self, request):
  918.         host = request.get_host()
  919.         if not host:
  920.             raise URLError('no host given')
  921.         
  922.         if request.has_data():
  923.             data = request.get_data()
  924.             if not request.has_header('Content-type'):
  925.                 request.add_unredirected_header('Content-type', 'application/x-www-form-urlencoded')
  926.             
  927.             if not request.has_header('Content-length'):
  928.                 request.add_unredirected_header('Content-length', '%d' % len(data))
  929.             
  930.         
  931.         (scheme, sel) = splittype(request.get_selector())
  932.         (sel_host, sel_path) = splithost(sel)
  933.         if not request.has_header('Host'):
  934.             if not sel_host:
  935.                 pass
  936.             request.add_unredirected_header('Host', host)
  937.         
  938.         for name, value in self.parent.addheaders:
  939.             name = name.capitalize()
  940.             if not request.has_header(name):
  941.                 request.add_unredirected_header(name, value)
  942.                 continue
  943.         
  944.         return request
  945.  
  946.     
  947.     def do_open(self, http_class, req):
  948.         '''Return an addinfourl object for the request, using http_class.
  949.  
  950.         http_class must implement the HTTPConnection API from httplib.
  951.         The addinfourl return value is a file-like object.  It also
  952.         has methods and attributes including:
  953.             - info(): return a mimetools.Message object for the headers
  954.             - geturl(): return the original request URL
  955.             - code: HTTP status code
  956.         '''
  957.         host = req.get_host()
  958.         if not host:
  959.             raise URLError('no host given')
  960.         
  961.         h = http_class(host)
  962.         h.set_debuglevel(self._debuglevel)
  963.         headers = dict(req.headers)
  964.         headers.update(req.unredirected_hdrs)
  965.         headers['Connection'] = 'close'
  966.         
  967.         try:
  968.             h.request(req.get_method(), req.get_selector(), req.data, headers)
  969.             r = h.getresponse()
  970.         except socket.error:
  971.             err = None
  972.             raise URLError(err)
  973.  
  974.         r.recv = r.read
  975.         fp = socket._fileobject(r)
  976.         resp = addinfourl(fp, r.msg, req.get_full_url())
  977.         resp.code = r.status
  978.         resp.msg = r.reason
  979.         return resp
  980.  
  981.  
  982.  
  983. class HTTPHandler(AbstractHTTPHandler):
  984.     
  985.     def http_open(self, req):
  986.         return self.do_open(httplib.HTTPConnection, req)
  987.  
  988.     http_request = AbstractHTTPHandler.do_request_
  989.  
  990. if hasattr(httplib, 'HTTPS'):
  991.     
  992.     class HTTPSHandler(AbstractHTTPHandler):
  993.         
  994.         def https_open(self, req):
  995.             return self.do_open(httplib.HTTPSConnection, req)
  996.  
  997.         https_request = AbstractHTTPHandler.do_request_
  998.  
  999.  
  1000.  
  1001. class HTTPCookieProcessor(BaseHandler):
  1002.     
  1003.     def __init__(self, cookiejar = None):
  1004.         if cookiejar is None:
  1005.             cookiejar = cookielib.CookieJar()
  1006.         
  1007.         self.cookiejar = cookiejar
  1008.  
  1009.     
  1010.     def http_request(self, request):
  1011.         self.cookiejar.add_cookie_header(request)
  1012.         return request
  1013.  
  1014.     
  1015.     def http_response(self, request, response):
  1016.         self.cookiejar.extract_cookies(response, request)
  1017.         return response
  1018.  
  1019.     https_request = http_request
  1020.     https_response = http_response
  1021.  
  1022.  
  1023. class UnknownHandler(BaseHandler):
  1024.     
  1025.     def unknown_open(self, req):
  1026.         type = req.get_type()
  1027.         raise URLError('unknown url type: %s' % type)
  1028.  
  1029.  
  1030.  
  1031. def parse_keqv_list(l):
  1032.     '''Parse list of key=value strings where keys are not duplicated.'''
  1033.     parsed = { }
  1034.     for elt in l:
  1035.         (k, v) = elt.split('=', 1)
  1036.         if v[0] == '"' and v[-1] == '"':
  1037.             v = v[1:-1]
  1038.         
  1039.         parsed[k] = v
  1040.     
  1041.     return parsed
  1042.  
  1043.  
  1044. def parse_http_list(s):
  1045.     '''Parse lists as described by RFC 2068 Section 2.
  1046.     
  1047.     In particular, parse comma-separated lists where the elements of
  1048.     the list may include quoted-strings.  A quoted-string could
  1049.     contain a comma.  A non-quoted string could have quotes in the
  1050.     middle.  Neither commas nor quotes count if they are escaped.
  1051.     Only double-quotes count, not single-quotes.
  1052.     '''
  1053.     res = []
  1054.     part = ''
  1055.     escape = quote = False
  1056.     for cur in s:
  1057.         if escape:
  1058.             part += cur
  1059.             escape = False
  1060.             continue
  1061.         
  1062.         if quote:
  1063.             if cur == '\\':
  1064.                 escape = True
  1065.                 continue
  1066.             elif cur == '"':
  1067.                 quote = False
  1068.             
  1069.             part += cur
  1070.             continue
  1071.         
  1072.         if cur == ',':
  1073.             res.append(part)
  1074.             part = ''
  1075.             continue
  1076.         
  1077.         if cur == '"':
  1078.             quote = True
  1079.         
  1080.         part += cur
  1081.     
  1082.     if part:
  1083.         res.append(part)
  1084.     
  1085.     return [ part.strip() for part in res ]
  1086.  
  1087.  
  1088. class FileHandler(BaseHandler):
  1089.     
  1090.     def file_open(self, req):
  1091.         url = req.get_selector()
  1092.         if url[:2] == '//' and url[2:3] != '/':
  1093.             req.type = 'ftp'
  1094.             return self.parent.open(req)
  1095.         else:
  1096.             return self.open_local_file(req)
  1097.  
  1098.     names = None
  1099.     
  1100.     def get_names(self):
  1101.         if FileHandler.names is None:
  1102.             FileHandler.names = (socket.gethostbyname('localhost'), socket.gethostbyname(socket.gethostname()))
  1103.         
  1104.         return FileHandler.names
  1105.  
  1106.     
  1107.     def open_local_file(self, req):
  1108.         import email.Utils as email
  1109.         host = req.get_host()
  1110.         file = req.get_selector()
  1111.         localfile = url2pathname(file)
  1112.         stats = os.stat(localfile)
  1113.         size = stats.st_size
  1114.         modified = email.Utils.formatdate(stats.st_mtime, usegmt = True)
  1115.         mtype = mimetypes.guess_type(file)[0]
  1116.         if not mtype:
  1117.             pass
  1118.         headers = mimetools.Message(StringIO('Content-type: %s\nContent-length: %d\nLast-modified: %s\n' % ('text/plain', size, modified)))
  1119.         if host:
  1120.             (host, port) = splitport(host)
  1121.         
  1122.         if (not host or not port) and socket.gethostbyname(host) in self.get_names():
  1123.             return addinfourl(open(localfile, 'rb'), headers, 'file:' + file)
  1124.         
  1125.         raise URLError('file not on local host')
  1126.  
  1127.  
  1128.  
  1129. class FTPHandler(BaseHandler):
  1130.     
  1131.     def ftp_open(self, req):
  1132.         host = req.get_host()
  1133.         if not host:
  1134.             raise IOError, ('ftp error', 'no host given')
  1135.         
  1136.         (host, port) = splitport(host)
  1137.         if port is None:
  1138.             port = ftplib.FTP_PORT
  1139.         else:
  1140.             port = int(port)
  1141.         (user, host) = splituser(host)
  1142.         if user:
  1143.             (user, passwd) = splitpasswd(user)
  1144.         else:
  1145.             passwd = None
  1146.         host = unquote(host)
  1147.         if not user:
  1148.             pass
  1149.         user = unquote('')
  1150.         if not passwd:
  1151.             pass
  1152.         passwd = unquote('')
  1153.         
  1154.         try:
  1155.             host = socket.gethostbyname(host)
  1156.         except socket.error:
  1157.             msg = None
  1158.             raise URLError(msg)
  1159.  
  1160.         (path, attrs) = splitattr(req.get_selector())
  1161.         dirs = path.split('/')
  1162.         dirs = map(unquote, dirs)
  1163.         dirs = dirs[:-1]
  1164.         file = dirs[-1]
  1165.         if dirs and not dirs[0]:
  1166.             dirs = dirs[1:]
  1167.         
  1168.         
  1169.         try:
  1170.             fw = self.connect_ftp(user, passwd, host, port, dirs)
  1171.             if not file or 'I':
  1172.                 pass
  1173.             type = 'D'
  1174.             for attr in attrs:
  1175.                 (attr, value) = splitvalue(attr)
  1176.                 if attr.lower() == 'type' and value in ('a', 'A', 'i', 'I', 'd', 'D'):
  1177.                     type = value.upper()
  1178.                     continue
  1179.             
  1180.             (fp, retrlen) = fw.retrfile(file, type)
  1181.             headers = ''
  1182.             mtype = mimetypes.guess_type(req.get_full_url())[0]
  1183.             if mtype:
  1184.                 headers += 'Content-type: %s\n' % mtype
  1185.             
  1186.             if retrlen is not None and retrlen >= 0:
  1187.                 headers += 'Content-length: %d\n' % retrlen
  1188.             
  1189.             sf = StringIO(headers)
  1190.             headers = mimetools.Message(sf)
  1191.             return addinfourl(fp, headers, req.get_full_url())
  1192.         except ftplib.all_errors:
  1193.             msg = None
  1194.             raise IOError, ('ftp error', msg), sys.exc_info()[2]
  1195.  
  1196.  
  1197.     
  1198.     def connect_ftp(self, user, passwd, host, port, dirs):
  1199.         fw = ftpwrapper(user, passwd, host, port, dirs)
  1200.         return fw
  1201.  
  1202.  
  1203.  
  1204. class CacheFTPHandler(FTPHandler):
  1205.     
  1206.     def __init__(self):
  1207.         self.cache = { }
  1208.         self.timeout = { }
  1209.         self.soonest = 0
  1210.         self.delay = 60
  1211.         self.max_conns = 16
  1212.  
  1213.     
  1214.     def setTimeout(self, t):
  1215.         self.delay = t
  1216.  
  1217.     
  1218.     def setMaxConns(self, m):
  1219.         self.max_conns = m
  1220.  
  1221.     
  1222.     def connect_ftp(self, user, passwd, host, port, dirs):
  1223.         key = (user, host, port, '/'.join(dirs))
  1224.         if key in self.cache:
  1225.             self.timeout[key] = time.time() + self.delay
  1226.         else:
  1227.             self.cache[key] = ftpwrapper(user, passwd, host, port, dirs)
  1228.             self.timeout[key] = time.time() + self.delay
  1229.         self.check_cache()
  1230.         return self.cache[key]
  1231.  
  1232.     
  1233.     def check_cache(self):
  1234.         t = time.time()
  1235.         if self.soonest <= t:
  1236.             for k, v in self.timeout.items():
  1237.                 if v < t:
  1238.                     self.cache[k].close()
  1239.                     del self.cache[k]
  1240.                     del self.timeout[k]
  1241.                     continue
  1242.             
  1243.         
  1244.         self.soonest = min(self.timeout.values())
  1245.         if len(self.cache) == self.max_conns:
  1246.             for k, v in self.timeout.items():
  1247.                 if v == self.soonest:
  1248.                     del self.cache[k]
  1249.                     del self.timeout[k]
  1250.                     break
  1251.                     continue
  1252.             
  1253.             self.soonest = min(self.timeout.values())
  1254.         
  1255.  
  1256.  
  1257.  
  1258. class GopherHandler(BaseHandler):
  1259.     
  1260.     def gopher_open(self, req):
  1261.         host = req.get_host()
  1262.         if not host:
  1263.             raise GopherError('no host given')
  1264.         
  1265.         host = unquote(host)
  1266.         selector = req.get_selector()
  1267.         (type, selector) = splitgophertype(selector)
  1268.         (selector, query) = splitquery(selector)
  1269.         selector = unquote(selector)
  1270.         if query:
  1271.             query = unquote(query)
  1272.             fp = gopherlib.send_query(selector, query, host)
  1273.         else:
  1274.             fp = gopherlib.send_selector(selector, host)
  1275.         return addinfourl(fp, noheaders(), req.get_full_url())
  1276.  
  1277.  
  1278.  
  1279. class OpenerFactory:
  1280.     default_handlers = [
  1281.         UnknownHandler,
  1282.         HTTPHandler,
  1283.         HTTPDefaultErrorHandler,
  1284.         HTTPRedirectHandler,
  1285.         FTPHandler,
  1286.         FileHandler]
  1287.     handlers = []
  1288.     replacement_handlers = []
  1289.     
  1290.     def add_handler(self, h):
  1291.         self.handlers = self.handlers + [
  1292.             h]
  1293.  
  1294.     
  1295.     def replace_handler(self, h):
  1296.         pass
  1297.  
  1298.     
  1299.     def build_opener(self):
  1300.         opener = OpenerDirector()
  1301.         for ph in self.default_handlers:
  1302.             if inspect.isclass(ph):
  1303.                 ph = ph()
  1304.             
  1305.             opener.add_handler(ph)
  1306.         
  1307.  
  1308.  
  1309.